from IO.input.src.make_input import *

NORMAL_CASES = 4
BAD_CASES = 1
TEST_CASES = NORMAL_CASES + BAD_CASES

INPUT_DIST = "IO/dist/"
SOLUTION_EXE = r"E:\workspace_Pycharm\Generate_Data_By_Cyaron\gen\IO\output\src\sol.exe"


def main():
    for i in range(1, TEST_CASES + 1):
        test_data = IO(file_prefix=INPUT_DIST, data_id=i)

        if 1 <= i <= NORMAL_CASES:
            make_input(test_data)
            print("normal %d.in" % i)
            test_data.output_gen(SOLUTION_EXE)
            print("normal %d.out" % i)
        else:
            make_bad_input(test_data)
            print("bad %d.in" % i)
            test_data.output_gen(SOLUTION_EXE)
            print("bad %d.out" % i)


if __name__ == '__main__':
    main()
